package com.itheima.ai.repository;

import com.itheima.model.ai.pojos.ChatMessage;
import org.springframework.data.mongodb.repository.MongoRepository;
import org.springframework.stereotype.Repository;

import java.util.List;

@Repository
public interface ChatMessageRepository extends MongoRepository<ChatMessage, String> {
    
    /**
     * 根据会话ID查询消息列表
     */
    List<ChatMessage> findBySessionIdOrderByCreateTimeAsc(String sessionId);
    
    /**
     * 根据用户ID和会话ID查询消息
     */
    List<ChatMessage> findByUserIdAndSessionIdOrderByCreateTimeAsc(Long userId, String sessionId);
    
    /**
     * 删除会话的所有消息
     */
    void deleteBySessionId(String sessionId);
    
    /**
     * 统计会话消息数量
     */
    long countBySessionId(String sessionId);
}

